# !/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
@作者 杨阳
@时间 2018-07-20
@说明 微信聊天工具类
@注意事项
"""
import os

import itchat

from settings import CHAT_LOGIN_QRCODE_PIC_URL
from util import LogUtil


def chat_login():
    """
    微信登录的方法
    :return:
    """
    pass


def chat_already_logged(uuid):
    """
    根据qruuid判断用户是否已经登录
    :param uuid:        生成二维语码的uuid
    :return:            是否已经登录
    """
    return itchat.check_login(uuid=uuid)


def chat_get_qruuid():
    """
    获得生成二维码的uuid
    :return:
    """
    return itchat.get_QRuuid()


def chat_get_qr_pic(uuid):
    """
    根据二维码的uuid获取二维码
    :param uuid:    二维码uuid
    :return:        UUID，也是图标名称去掉后缀
    """
    LogUtil.info("=====================开始生成登录二维码=====================")
    LogUtil.info("二维码UUID:{}".format(uuid))

    pic_dir = CHAT_LOGIN_QRCODE_PIC_URL + uuid + ".png"
    LogUtil.info("二维码图片路径:{}".format(pic_dir))

    if not os.path.exists(CHAT_LOGIN_QRCODE_PIC_URL):
        os.makedirs(CHAT_LOGIN_QRCODE_PIC_URL)

    itchat.get_QR(uuid=uuid, picDir=pic_dir)
    return uuid


if __name__ == '__main__':
    itchat.auto_login(hotReload=True)
    itchat.run()
